import tensorflow as tf

layers = tf.keras.layers

x = tf.random.normal([2, 5, 5, 3])
layer = layers.Conv2D(4, kernel_size=3, strides=1, padding='SAME')
#print(layer.trainable_variables)
out = layer(x)
# print(out)
#print(layer.trainable_variables)
